from main import *
# 导入必要的模块
import os
import sys
import psutil
import time
import numpy as np
import multiprocessing as mp
import torch
# 定义一个函数，用于打印进程信息和 numpy 模块信息
def print_info():
    import main
    import torch
    # 获取当前进程对象和进程 id
    process = psutil.Process(os.getpid())
    pid = process.pid
    # 获取当前进程的内存占用
    memory = process.memory_info().rss / 1024 / 1024
if __name__ == '__main__':
   # 创建两个子进程，并在其中分别导入 numpy 模块，并打印信息
   a=time.time()
   p1 = mp.Process(target=print_info)
   p2 = mp.Process(target=print_info)
   p1.start()
   p2.start()
   p1.join()
   p2.join()
   print(time.time()-a)
